import pymysql

# 建立数据库连接


def main():
    # 连接数据库
    connect = pymysql.connect(host="39.107.245.102",port=3306,user="root",passwd="wuji5201314.",db="school",charset="utf8")
    # 查看连接是否完成
    # 建立游标连接，并使其使用完成自动删除
    try:
        with connect.cursor() as cursor:
            result=cursor.execute("insert into tb_student values (1011,'玄冥二老','0','1970-01-05','老远了','1')")
            print(result)
            # 这个返回数据返回的是几行收到影响，如果1则成功，0则失败，看你实际要改的库
            if result == 1:
                print('添加成功')
            else:
                print('添加失败')
    # 如果只到这里，数据是加不进来的，这只是一个事务，所以要包含成功失败的场景，即提交与回滚，
    # 所以需要加异常捕捉
            connect.commit()
    except pymysql.MySQLError as error:
        print('添加失败.{}'.format(error))
        connect.rollback()
    # finally:正常异常都执行，必须要释放资源


    try:
        # 获得游标对象
        with connect.cursor() as cursor:
            # 执行sql
            result=cursor.execute("delete from tb_student where stu_id=1011")
            print(result)
            if result == 1:
                print('删除成功')
            else:
                print('删除失败')
            # 提交事务
            connect.commit()
    except pymysql.MySQLError as error:
        print(error)
        # 事务失败回滚
        connect.rollback()
    finally:
        # 关闭链接释放资源
        connect.close()

    print(connect)
if __name__=="__main__":
    main()